home *** CD-ROM | disk | FTP | other *** search
/ Night Owl 6 / Night Owl's Shareware - PDSI-006 - Night Owl Corp (1990).iso / 039a / mawk.zip / MAIN.C < prev    next >
C/C++ Source or Header  |  1991-04-22  |  3KB  |  152 lines
  1.  
  2. /********************************************
  3. main.c
  4. copyright 1991, Michael D. Brennan
  5.  
  6. This is a source file for mawk, an implementation of
  7. the Awk programming language as defined in
  8. Aho, Kernighan and Weinberger, The AWK Programming Language,
  9. Addison-Wesley, 1988.
  10.  
  11. See the accompaning file, LIMITATIONS, for restrictions
  12. regarding modification and redistribution of this
  13. program in source or binary form.
  14. ********************************************/
  15.  
  16.  
  17. /* $Log:    main.c,v $
  18.  * Revision 2.2  91/04/22  08:08:58  brennan
  19.  * cannot close(3) or close(4) because of bug in TurboC++ 1.0
  20.  * 
  21.  * Revision 2.1  91/04/08  08:23:27  brennan
  22.  * VERSION 0.97
  23.  * 
  24. */
  25.  
  26.  
  27.  
  28. /*  main.c  */
  29.  
  30. #include "mawk.h"
  31. #include "code.h"
  32. #include "init.h"
  33. #include "fin.h"
  34. #include "bi_vars.h"
  35. #include "field.h"
  36. #include "files.h"
  37. #include <stdio.h>
  38.  
  39. #if  DOS 
  40. void  reargv(int *, char ***) ;
  41. #endif
  42.  
  43.  
  44. void  PROTO( process, (void) ) ;
  45. void  PROTO( main_loop, (void) ) ;
  46.  
  47. extern int program_fd ;
  48. char *progname ;
  49.  
  50. jmp_buf   exit_jump, next_jump ;
  51. int  exit_code ;
  52.  
  53.  
  54. main(argc , argv )
  55.   int argc ; char **argv ;
  56. { 
  57.  
  58. #if   DOS
  59.   progname = "mawk" ;
  60. #if      HAVE_REARGV
  61.   reargv(&argc, &argv) ;
  62. #endif
  63. #else
  64.   { char *strrchr() ;
  65.     char *p = strrchr(argv[0], '/') ;
  66.     progname = p ? p+1 : argv[0] ; }
  67. #endif
  68.  
  69.   initialize(argc, argv) ;
  70.  
  71.   if ( parse() || compile_error_count )  exit(1) ;
  72.  
  73.   compile_cleanup() ;
  74.   process() ;
  75.  
  76.   mawk_exit( exit_code ) ;
  77.   return 0 ;
  78. }
  79.  
  80.  
  81. static  void  process()
  82. {
  83.  
  84.   if ( setjmp(exit_jump) )
  85.   { if ( begin_start ) zfree(begin_start, begin_size) ;
  86.     goto the_exit ;
  87.   }
  88.  
  89.   if ( begin_start )
  90.   { (void) execute(begin_start, eval_stack-1, 0) ;
  91.     zfree( begin_start , begin_size ) ;
  92.     begin_start = (INST *) 0 ;
  93.   }
  94.  
  95.   if ( main_start || end_start )  main_loop() ;
  96.  
  97. the_exit:
  98.  
  99.   if ( setjmp(exit_jump) )  mawk_exit(exit_code) ;
  100.  
  101.   if ( main_start )  zfree(main_start, main_size) ;
  102.   if ( end_start ) (void) execute(end_start, eval_stack-1, 0) ;
  103. }
  104.  
  105.  
  106. static void  main_loop()
  107. { register char *p ;
  108.   unsigned len ;
  109.  
  110.   /* the main file stream might already be open by a call of
  111.      getline in the BEGIN block */
  112.  
  113.   if ( main_fin == (FIN *) -1 && ! open_main()
  114.        || ! main_fin )  return ;
  115.  
  116.   if ( main_start )
  117.   {
  118.      (void)  setjmp(next_jump) ;
  119.  
  120.      while ( p = FINgets( main_fin, &len ) )
  121.      { 
  122.        if ( TEST2(bi_vars + NR) != TWO_DOUBLES )
  123.                     cast2_to_d(bi_vars + NR) ;
  124.  
  125.        bi_vars[NR].dval += 1.0 ;
  126.        bi_vars[FNR].dval += 1.0 ;
  127.  
  128.        set_field0(p, len) ;
  129.        (void) execute( main_start, eval_stack-1, 0) ;
  130.      }
  131.   }
  132.   else  /* eat main to set NR and FNR before executing END */
  133.   { long nr ;
  134.  
  135.     if ( TEST2(bi_vars+NR) != TWO_DOUBLES ) cast2_to_d(bi_vars+NR) ;
  136.     nr = (long) bi_vars[NR].dval ;
  137.     while ( FINgets( main_fin, &len ) )
  138.     { nr++ ; bi_vars[FNR].dval += 1.0 ; }
  139.     bi_vars[NR].dval = (double) nr ;
  140.   }
  141. }
  142.  
  143.  
  144. void  mawk_exit(x)
  145.   int x ;
  146. {
  147. #if  !  DOS
  148.   close_out_pipes() ;  /* no effect, if no out pipes */
  149. #endif
  150.   exit(x) ;
  151. }
  152.